package com.moyoutian.leetcode.jianzhi.general;

/**
 * 剑指 Offer 22. 链表中倒数第k个节点
 * <p>
 * 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。
 * <p>
 * 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
 * <p>
 * <p>
 * <p>
 * 示例：
 * <p>
 * 给定一个链表: 1->2->3->4->5, 和 k = 2.
 * <p>
 * 返回链表 4->5.
 */
public class Demo22 {


    public static class ListNode {
        int val;
        ListNode next;

        ListNode(int x) {
            val = x;
        }
    }

    public static void main(String[] args) {

        int[] vals = {1, 2, 3, 4, 5};

        ListNode header = new ListNode(vals[0]);
        ListNode tempNode = header;
        for (int i = 1; i < vals.length; i++) {
            tempNode.next = new ListNode(vals[i]);
            tempNode = tempNode.next;
        }
        System.out.println(getKthFromEnd(header, 5).val);
    }

    /**
     * 第三种解法 ，双指针,一个先位移到底n-k步，然后在增加另一个从头开始的指针。始终和位移到n-k的指针保持k的距离。然后让首个指针位移到终点，那么另一个指针就是导数第k个
     *
     * @param head
     * @param k
     * @return
     */
    private static ListNode getKthFromEnd(ListNode head, int k) {

        //距离结束还有k个数
        ListNode temp = head;

        //假设一共有n个数，k=n-m
        int tempK = k - 1;
        ListNode tempKNode = head;
        while (temp != null) {
            if (tempK < 0) {
                tempKNode = tempKNode.next;
            }
            temp = temp.next;
            tempK--;
        }
        return tempKNode;
    }

    /**
     * 递归
     *
     * @param head
     * @param k
     * @return
     */
    public static ListNode getKthFromEnd2(ListNode head, int k) {
        //第二种想法，递归次数返回。总深度-当前深度=k,关键就是总深度怎么算。能同时不等于当前深度。这里咱们用引用类型
        int[] all = new int[1];
        ListNode[] node = new ListNode[1];
        getKthFromEnd2(head, k, all, 0, node);
        return node[0];
    }

    /**
     * 总深度-当前深度=k
     *
     * @param all  总深度
     * @param curr 当前深度
     * @param node 返回结点
     */
    public static void getKthFromEnd2(ListNode head, int k, int[] all, int curr, ListNode[] node) {
        //第二种想法，递归次数返回。总深度-当前深度=k
        //先用递归算总深度
        if (head != null) {
            ++all[0];
            getKthFromEnd2(head.next, k, all, ++curr, node);
        }
        //在算完总深度后判断结点是否是返回结点
        if (all[0] - curr == k - 1) {
            node[0] = head;
        }
    }

    /**
     * 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。
     * 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
     *
     * @param head
     * @param k
     * @return
     */
    public static ListNode getKthFromEnd1(ListNode head, int k) {

        //第一种想法，先算出一共有多少个结点，然后在算出在正数第几个。在正序便利返回
        int nodeCount = getNodeCount(head);
        int index = nodeCount - k;
        ListNode tempNode = head;
        while (index > 0 && tempNode != null) {
            tempNode = tempNode.next;
            index--;
        }
        return tempNode;
    }

    public static int getNodeCount(ListNode head) {
        int count = 0;
        ListNode tempNode = head;
        while (tempNode != null) {
            tempNode = tempNode.next;
            count++;
        }
        return count;
    }
}
